# !/usr/bin/python3
# -*- coding: utf-8 -*-
# @Author : Lawrence
# @File : insert_pdf.py
# @Project : worktool
# @Time : 2024/11/11 18:38
import os

import PyPDF2


def insert_pdf_into_pdf(source_path, insert_path, output_path, page_index):
    # 打开源 PDF 文件
    source_pdf = PyPDF2.PdfReader(source_path)
    # 打开要插入的 PDF 文件
    insert_pdf = PyPDF2.PdfReader(insert_path)

    # 创建一个新的 PDF 写入对象
    output_pdf = PyPDF2.PdfWriter()

    # 将源 PDF 的前 part1 页复制到输出 PDF
    for i in range(page_index):
        output_pdf.add_page(source_pdf.pages[i])

    # 将要插入的 PDF 的所有页面复制到输出 PDF
    for page in insert_pdf.pages:
        output_pdf.add_page(page)

    # 将源 PDF 的剩余部分复制到输出 PDF
    for i in range(page_index, len(source_pdf.pages)):
        output_pdf.add_page(source_pdf.pages[i])

    # 将输出 PDF 写入到文件
    with open(output_path, 'wb') as output_file:
        output_pdf.write(output_file)


# 示例用法
source_path = './data/output1.pdf'
insert_path = './data/2024-JLDJCA-F4005某单位办公助手-AI辅助生成系统(1).pdf'
output_path = 'data/output2.pdf'
page_index = 31  # 在第 3 页之前插入 insert.pdf

insert_pdf_into_pdf(source_path, insert_path, output_path, page_index)